# os 模块

import os
import xlrd2


def file_is_exist(file_path):
    """
    判断文件是否存在
    :param file_path: 文件路径
    :return:
    """
    if os.path.isfile(file_path):
        return True
    else:
        raise Exception("文件未找到")


def get_file_data(file_path):
    """
    读取 txt 记事本中的数据
    :param file_path:
    :return:
    """
    file_is_exist(file_path)
    file = open(file=file_path, mode='r', encoding='utf-8')
    result = []
    for data in file:
        result.append(tuple(data.strip().split(',')))
    file.close()
    return result


def get_file_excel_data(file_path):
    """
    读取 excel 中的数据
    想要操作 excel   需要下载 xlrd 模块 pip install xlrd2
    :param file_path:
    :return:
    """
    file_is_exist(file_path)
    book = xlrd2.open_workbook("data.xlsx")
    sh = book.sheet_by_index(0)
    result = []
    for rx in range(sh.nrows):
        result.append(tuple(sh.row_values(rx)))
    return result


if __name__ == '__main__':
    print(get_file_excel_data('data.xlsx'))
